import os

def batch_rename(folder_path, prefix):
    # 获取文件夹中的所有文件
    files = os.listdir(folder_path)
    
    # 过滤掉文件夹中的子文件夹
    files = [f for f in files if os.path.isfile(os.path.join(folder_path, f))]
    
    # # 对文件进行排序（可选）
    # files.sort()
    
    # 遍历所有文件并重命名
    for i, filename in enumerate(files):
        # 分离文件名和扩展名
        file_root, file_ext = os.path.splitext(filename)
        
        # 构建新的文件名
        new_filename = f"{prefix}_{i+1}{file_ext}"
        
        # 获取旧文件路径和新文件路径
        old_file_path = os.path.join(folder_path, filename)
        new_file_path = os.path.join(folder_path, new_filename)
        
        # 重命名文件
        os.rename(old_file_path, new_file_path)
        # print(f"Renamed: {old_file_path} -> {new_file_path}")

if __name__ == "__main__":
    # 设置文件夹路径和前缀
    folder_path = "./output"
    prefix = "嘻嘻"
    
    # 调用批量重命名函数
    batch_rename(folder_path, prefix)
